/**
 * 给定一个由表示变量之间关系的字符串方程组成的数组，每个字符串方程 equations[i] 的长度为 4，并采用两种不同的形式之一："a==b" 或 "a!=b"。在这里，a 和 b 是小写字母（不一定不同），表示单字母变量名。
 *
 * 只有当可以将整数分配给变量名，以便满足所有给定的方程时才返回 true，否则返回 false。
 * https://leetcode.cn/problems/satisfiability-of-equality-equations/description/
 */
class EquationsPossible {
    public boolean equationsPossible(String[] equations) {
        List<String> list = new ArrayList<>();
        UnionFindSet unionFindSet = new UnionFindSet(26);
        for(int i=0;i<equations.length;i++) {
            char ch = equations[i].charAt(1);
            if(ch=='=') {
                unionFindSet.union(equations[i].charAt(0)-'a',equations[i].charAt(3)-'a');
            } else {
                list.add(equations[i]);
            }
        }
        for(int i=0;i<list.size();i++) {
            String str = list.get(i);
            int x1 = str.charAt(0)-'a';
            int x2 = str.charAt(3)-'a';
            if(unionFindSet.isSameUnionFindSet(x1,x2)) {
                return false;
            }
        }

        return true;
    }
}
 class UnionFindSet {
    public int[] elem;
    public int usedSize;

    public UnionFindSet(int n) {
        this.elem = new int[n];
        Arrays.fill(elem,-1);
    }

    /**
     * 查找数据x的根节点
     * @param x
     * @return
     */
    public int findRoot(int x) {
        if(x<0) {
            throw new IndexOutOfBoundsException("下标不合法，是负数");
        }
        while(elem[x]>=0) {
            x = elem[x];
        }
        return x;
    }

    /**
     * 判断x1和x2是不是同一个集合
     * @param x1
     * @param x2
     * @return
     */
    public boolean isSameUnionFindSet(int x1,int x2) {
        int index1= findRoot(x1);
        int index2 = findRoot(x2);
        return index1==index2;
    }

    /**
     * 合并操作
     * @param x1
     * @param x2
     */
    public void union(int x1,int x2) {
        int index1 = findRoot(x1);
        int index2 = findRoot(x2);
        if(index1==index2) {
            return;
        }
        elem[index1] = elem[index1]+elem[index2];
        elem[index2] = index1;
    }

    /**
     *
     * @return
     */
    public int getCount() {
        int count =0;
        for(int x:elem) {
            if(x<0) {
                count++;
            }
        }
        return count;
    }
    
}